#pragma once
#include <iostream>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
class InetAddr
{
public:
    //有参构造，sockaddr_in 构造
    InetAddr(const sockaddr_in& addr)
    :_addr(addr)
    {
        get_address(&_ip,&_port);
    }
    //有参构造，ip 和 port 构造
    InetAddr(const std::string& ip,uint16_t port)
    :_ip(ip)
    ,_port(port)
    {
        _addr.sin_family = AF_INET;
        _addr.sin_port = htons(_port);
        _addr.sin_addr.s_addr = inet_addr(_ip.c_str());
    }
    //默认构造
    InetAddr()
    {}
    //返回ip
    std::string Ip()
    {
        return _ip;
    }
    //返回：端口号
    uint16_t Port()
    {
        return _port;
    }
    //判断相等
    bool operator==(const InetAddr& addr) 
    {
        if(_ip == addr._ip&&_port == addr._port)
        {
            return true;
        }
        else
        return false;
    }
    //返回sockaddr_in
    struct sockaddr_in addr()
    {
        return _addr;
    }
    //析构函数
    ~InetAddr()
    {}
private:
    //从_addr 中获取ip 和 port 信息
    void get_address(std::string* ip,uint16_t* port)
    {
        *ip = inet_ntoa(_addr.sin_addr);
        *port = ntohs(_addr.sin_port);
    }
    struct sockaddr_in _addr;
    std::string _ip;
    uint16_t _port;
};